nums = [1, 3, 4, 2, 8]
new_nums = [n*n for n in nums if n < 5]

print(new_nums)
